package main

import (
	"fmt"
	"github.com/hashicorp/consul/api"
	"go.uber.org/zap"
	"google.golang.org/grpc"
	"google.golang.org/grpc/health"
	"google.golang.org/grpc/health/grpc_health_v1"
	"net"
	"os"
	"os/signal"
	"syscall"
	"user_srv/global"
	"user_srv/handler"
	_ "user_srv/initialize"
	user_pb "user_srv/user"
)

func main() {
	zap.S().Info(global.ServerConfig)
	//注册微服务
	g := grpc.NewServer()
	//创建一个server实例
	s := handler.UserService{}
	//在grpc上注册服务
	user_pb.RegisterUserServer(g, &s)

	//注册服务健康状态检查
	grpc_health_v1.RegisterHealthServer(g, health.NewServer())

	//监听指定的网络地址和端口

	listen, err := net.Listen("tcp", fmt.Sprintf("%s:%d", global.ServerConfig.Ip, global.ServerConfig.Port))
	if err != nil {
		zap.S().Panic(err)
	}

	//启动grpc服务器并开始监听
	go func() {
		err := g.Serve(listen)
		if err != nil {
			panic("failed to start grpc:" + err.Error())
		}
	}()
	//如果用户微服务注销了，就应该服务注册发现也注销了
	// 注销服务
	// 等待中断信号以优雅地关闭服务器（设置 5 秒的超时时间）
	//接收终止信号
	quit := make(chan os.Signal)
	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
	<-quit

	config := api.DefaultConfig()
	config.Address = fmt.Sprintf("%s:%s", global.ServerConfig.ConsulConfig.ConsulHost, global.ServerConfig.ConsulConfig.ConsulPort)

	//实例化consul客户端
	client, err := api.NewClient(config)
	if err != nil {
		zap.S().Panic(err)
	}

	err = client.Agent().ServiceDeregister(global.ServerId)
	if err != nil {
		zap.S().Info("服务注销失败")
		return
	}

	zap.S().Info("服务注销成功")
}
